Update app.py

app.py

@@ -12,17 +12,19 @@ def generate_haiku_comment(code):
     """
     Generates a haiku comment for the given code snippet.
     """
-    prompt = (
-        f"Analyze the following code and write a poetic haiku as a comment summarizing its purpose or essence:\n\n"
-        f"{code}\n\n"
-        "Focus on creativity and capturing the core idea."
+   prompt = (
+        f"Analyze the following code and perform two tasks:\n\n"
+        f"1. Write a poetic haiku as a comment summarizing its purpose or essence.\n"
+        f"2. Refactor and clean the code while maintaining its functionality.\n\n"
+        f"Code:\n{code}\n\n"
+        "Provide the output as haiku comment followed by the refactored code."
     )
 
     try:
         response = client.chat.completions.create(
             model='Qwen2.5-Coder-32B-Instruct',
             messages=[
-                {"role": "system", "content": "You are a creative haiku generator for code comments."},
+                {"role": "system", "content": "You are a creative haiku generator and code refactoring assistant."},
                 {"role": "user", "content": prompt}
             ],
             temperature=0.7,
@@ -37,8 +39,10 @@ def generate_haiku_comment(code):
 def gradio_interface():
     interface = gr.Interface(
         fn=generate_haiku_comment,
-        inputs=gr.Textbox(label="Code Snippet"),
-        outputs=gr.Textbox(label="Generated Haiku Comment"),
+        inputs=gr.Textbox(label="Code Snippet",lines=10, placeholder="Paste your code snippet here..."),
+        outputs= gr.Markdown(    """
+    Code will be generated here
+    """),
         title="Code Haiku Comment Generator",
         description="Paste a code snippet, and let the AI generate a poetic haiku comment summarizing it!",
     )