Update app.py

app.py

@@ -8,38 +8,39 @@ client = openai.OpenAI(
     base_url="https://api.sambanova.ai/v1",
 )
 
-def generate_code_haiku(theme):
+def generate_haiku_comment(code):
     """
-    Generates a code-themed haiku based on the input theme.
+    Generates a haiku comment for the given code snippet.
     """
     prompt = (
-        f"Write a poetic haiku about coding with the theme: '{theme}'. "
-        "Focus on creativity and programming metaphors."
+        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."
     )
 
     try:
         response = client.chat.completions.create(
-            model='Meta-Llama-3.1-8B-Instruct',
+            model='Qwen2.5-Coder-32B-Instruct',
             messages=[
-                {"role": "system", "content": "You are a creative haiku generator for coding themes."},
+                {"role": "system", "content": "You are a creative haiku generator for code comments."},
                 {"role": "user", "content": prompt}
             ],
             temperature=0.7,
             top_p=0.9
         )
-        haiku = response.choices[0].message.content.strip()
-        return haiku
+        haiku_comment = response.choices[0].message.content.strip()
+        return haiku_comment
     except Exception as e:
-        return f"Error generating haiku: {str(e)}"
+        return f"Error generating haiku comment: {str(e)}"
 
 # Gradio Interface
 def gradio_interface():
     interface = gr.Interface(
-        fn=generate_code_haiku,
-        inputs=gr.Textbox(label="Theme (e.g., Debugging, Algorithms, AI, etc.)"),
-        outputs=gr.Textbox(label="Generated Code Haiku"),
-        title="Code Haiku Generator",
-        description="Generate poetic haikus about programming themes. Enter a theme, and let the AI craft a haiku for you!",
+        fn=generate_haiku_comment,
+        inputs=gr.Textbox(label="Code Snippet"),
+        outputs=gr.Textbox(label="Generated Haiku Comment"),
+        title="Code Haiku Comment Generator",
+        description="Paste a code snippet, and let the AI generate a poetic haiku comment summarizing it!",
     )
     return interface