Update app.py

app.py

@@ -1,12 +1,16 @@
 import os
+import logging
 from groq import Groq
-from flask import Flask, render_template
+from flask import Flask
 from dotenv import load_dotenv
 
 load_dotenv()
 
 app = Flask(__name__)
 
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+
 # Set Groq API key
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
@@ -14,31 +18,39 @@ client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 model = "llama3-groq-70b-8192-tool-use-preview"
 
 # Set fixed prompt
-fixed_prompt = "generate a concise silly letter to parents about their child detention reasons. write extreme behaviours"
+fixed_prompt = "generate a silly letter to parents about their child detention reasons. write extreme behaviours"
 
 # Define function to generate text
 def generate_text():
-    completion = client.chat.completions.create(
-        model=model,
-        messages=[
-            {
-                "role": "user",
-                "content": fixed_prompt
-            }
-        ],
-        temperature=0.5,
-        max_tokens=1024,
-        top_p=0.65,
-        stream=False,
-        stop=None,
-    )
-    return completion.choices[0].message.content.strip()
+    try:
+        completion = client.chat.completions.create(
+            model=model,
+            messages=[
+                {
+                    "role": "user",
+                    "content": fixed_prompt
+                }
+            ],
+            temperature=0.5,
+            max_tokens=1024,
+            top_p=0.65,
+            stream=False,
+            stop=None,
+        )
+        return completion.choices[0].message.content.strip()
+    except Exception as e:
+        app.logger.error(f"Error generating text: {str(e)}")
+        return "An error occurred while generating the text."
 
 # Define Flask route
 @app.route("/")
 def home():
-    message = generate_text()
-    return render_template("index.html", message=message)
+    try:
+        message = generate_text()
+        return f"<pre>{message}</pre>"
+    except Exception as e:
+        app.logger.error(f"Error in home route: {str(e)}")
+        return "An error occurred. Please try again later.", 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=False)
+    app.run(host="0.0.0.0", port=7860)