Update app.py

app.py

@@ -18,17 +18,24 @@ from gradio_client import Client
 from gradio_client import Client
 model_client = Client("https://tonic-surerag.hf.space/--replicas/m666j/")
 
-def generate_and_tweet(query):
+def generate_and_tweet(query):def generate_and_tweet(query):
     # Generate text based on the user's query
-    result = model_client.predict(query, api_name="/predict")
+    response = model_client.predict(query, api_name="/predict")
     
-    # Cut the text to fit Twitter's character limit if necessary
-    tweet_text = result[:180]  # Twitter's character limit is 280
+    # Assuming the response is a dictionary and the generated text is under a key, e.g., 'generated_text'
+    # Adjust the key as per your model's response structure
+    if isinstance(response, dict) and 'generated_text' in response:
+        tweet_text = response['generated_text']
+    elif isinstance(response, str):  # Direct string response
+        tweet_text = response
+    else:
+        return "Error: The model's response was not in the expected format."
     
-    # Post the generated text to Twitter
-    response = clientw.create_tweet(text=tweet_text)
+    # Ensure the tweet text is a string and trim it to Twitter's character limit
+    tweet_text = str(tweet_text)[:280]  # Twitter's character limit is 280
     
-    # Return the tweet URL or confirmation
+    tweet_text = result[:180]  # Twitter's character limit is 280
+    response = clientw.create_tweet(text=tweet_text)
     tweet_id = response.data['id']
     tweet_url = f"https://twitter.com/user/status/{tweet_id}"
     return f"Tweeted successfully! Here's the URL: {tweet_url}"