Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import io
 from PIL import Image
 import os
-import time  # Importing time to add delays for sequential execution
+import time
 
 # Set up your OpenAI API key (make sure it's stored as an environment variable)
 openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -28,20 +28,20 @@ else:
 
 API_URL = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
 
-# Define the OpenAI GPT-3 text generation function with error handling
-def generate_with_gpt3(prompt, max_tokens=150, temperature=0.7):
+# Define the OpenAI ChatCompletion function using `gpt-3.5-turbo`
+def generate_with_gpt3(prompt):
     try:
-        print("Generating text with GPT-3...")
-        response = openai.Completion.create(
-            engine="text-davinci-003",  # Use "text-davinci-003" for high-quality outputs
-            prompt=prompt,
-            max_tokens=max_tokens,
-            temperature=temperature,
+        print("Generating text with OpenAI ChatCompletion...")
+        # Use ChatCompletion with gpt-3.5-turbo
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # Use "gpt-4" if you have access
+            messages=[{"role": "system", "content": "You are a helpful assistant."},
+                      {"role": "user", "content": prompt}],
+            max_tokens=150,
+            temperature=0.2,
             top_p=0.9,
-            frequency_penalty=0.0,
-            presence_penalty=0.0
         )
-        generated_text = response.choices[0].text.strip()
+        generated_text = response['choices'][0]['message']['content'].strip()
         print("Text generation completed.")
         return generated_text
     except Exception as e:
@@ -61,20 +61,18 @@ def translate_and_generate_image(tamil_text):
     except Exception as e:
         return "Error during translation: " + str(e), "", None
 
-    # Ensure sequential flow by waiting before moving to the next step
-    time.sleep(1)  # Optional: Add a small delay to ensure proper execution order
+    time.sleep(1)  # Optional: Small delay to ensure sequential execution
 
-    # Step 2: Generate high-quality descriptive text using OpenAI's GPT-3
+    # Step 2: Generate high-quality descriptive text using OpenAI's ChatCompletion
     try:
         print("Generating descriptive text from translated English text...")
         prompt = f"Create a detailed and creative description based on the following text: {translated_text}"
-        generated_text = generate_with_gpt3(prompt, max_tokens=150, temperature=0.7)
+        generated_text = generate_with_gpt3(prompt)
         print(f"Text generation completed: {generated_text}")
     except Exception as e:
         return translated_text, f"Error during text generation: {e}", None
 
-    # Ensure sequential flow by waiting before moving to the next step
-    time.sleep(1)  # Optional: Add a small delay to ensure proper execution order
+    time.sleep(1)  # Optional: Small delay to ensure sequential execution
 
     # Step 3: Use the generated English text to create an image
     try:
@@ -101,7 +99,7 @@ iface = gr.Interface(
              gr.Textbox(label="Generated Descriptive Text"),
              gr.Image(label="Generated Image")],
     title="Tamil to English Translation, GPT-3 Text Generation, and Image Creation",
-    description="Translate Tamil text to English using Facebook's mbart-large-50 model, generate high-quality text using GPT-3, and create an image using the generated text.",
+    description="Translate Tamil text to English using Facebook's mbart-large-50 model, generate high-quality text using GPT-3.5-turbo, and create an image using the generated text.",
 )
 
 # Launch Gradio app without `share=True`