Update app.py

app.py

@@ -1,6 +1,36 @@
 from huggingface_hub import InferenceClient
 import gradio as gr
 
+import re
+
+# Define regex patterns for comments
+java_single_line_comment_regex = r"\/\/.*"
+java_multiline_comment_regex = r"\/\*(?:[^*]|\*(?!\/))*\*\/"
+kotlin_single_line_comment_regex = r"\/\/.*"
+kotlin_multiline_comment_regex = r"\/\*(?:[^*]|\*(?!\/))*\*\/"
+
+
+def remove_comments(content,file_type):
+    """
+    Opens a Java or Kotlin file, removes comments, and saves the changes.
+
+    Args:
+        file_path: The path to the Java or Kotlin file.
+    """
+
+
+    # Determine file type based on extension
+    if file_type == "java":
+        pattern = java_single_line_comment_regex + "|" + java_multiline_comment_regex
+    elif file_type == "kotlin":
+        pattern = kotlin_single_line_comment_regex + "|" + kotlin_multiline_comment_regex
+    else:
+        raise ValueError(f"Unsupported file type: {file_type}")
+
+    # Remove comments using regex
+    clean_content = re.sub(pattern, "", content, )
+    return clean_content
+    
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 def format_prompt(message, history):
@@ -27,7 +57,11 @@ def generate(
         do_sample=True,
         seed=42,
     )
-
+    prompt = f"""Translate the given Kotlin code to Java, adhering to the following constraints:
+Preserve the original names of classes, fields, and methods without renaming.
+{remove_comments(
+    content=prompt,file_type='kotlin'
+).strip()}"""
     formatted_prompt = format_prompt(prompt, history)
 
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
@@ -44,7 +78,7 @@ mychatbot = gr.Chatbot(
 
 demo = gr.ChatInterface(fn=generate, 
                         chatbot=mychatbot,
-                        title="Tomoniai's Mixtral 8x7b Chat",
+                        title="Mixtral 8x7b Chat For Kotlin Translation",
                         retry_btn=None,
                         undo_btn=None
                        )