Update app.py

app.py

@@ -1,43 +1,33 @@
 import gradio as gr
-import requests
-import urllib.parse
+import http.client
 from PIL import Image, ImageDraw, ImageFont
+import urllib.parse
 
-# RapidAPI Credentials and API Endpoint
+# RapidAPI credentials and host
 API_KEY = "2e427e3d07mshba1bdb10cb6eb30p12d12fjsn215dd7746115"  # Replace with your actual API key
 API_HOST = "horoscopes-ai.p.rapidapi.com"
-API_URL_TEMPLATE = "https://horoscopes-ai.p.rapidapi.com/get_horoscope/{sign}/{period}/general/en"
 
-# Function to Fetch Horoscope and Create Twitter Share Link
+# Function to fetch horoscope based on sign and period
 def get_horoscope(sign, period="today"):
-    url = API_URL_TEMPLATE.format(sign=sign, period=period)
+    conn = http.client.HTTPSConnection(API_HOST)
     
     headers = {
-        "x-rapidapi-key": API_KEY,
-        "x-rapidapi-host": API_HOST
+        'x-rapidapi-key': API_KEY,
+        'x-rapidapi-host': API_HOST
     }
     
-    try:
-        response = requests.get(url, headers=headers)
-        response.raise_for_status()
-        
-        # Parse JSON response and retrieve the horoscope text
-        data = response.json()
-        horoscope_text = data.get("general", ["No horoscope available"])[0]
-        
-        # Prepare the Twitter share link
-        share_text = f"Here's my horoscope for {sign.capitalize()} on {period.capitalize()}:\n{horoscope_text}"
-        encoded_text = urllib.parse.quote(share_text)  # URL-encode the text
-        twitter_share_url = f"https://twitter.com/intent/tweet?text={encoded_text}"
-        
-        # Return both the horoscope and the Twitter link
-        return horoscope_text, twitter_share_url
-    except requests.exceptions.RequestException as e:
-        return f"Error retrieving horoscope: {e}", None
+    # Construct the endpoint with the selected sign and period
+    endpoint = f"/get_horoscope_en/{sign}/{period}/general"
+    
+    # Make the GET request to the endpoint
+    conn.request("GET", endpoint, headers=headers)
+    res = conn.getresponse()
+    data = res.read()
+    return data.decode("utf-8")
 
-# Function to Generate Image from Horoscope Text
+# Function to generate an image from the horoscope text
 def generate_horoscope_image(text):
-    # Define image size and background color
+    # Set image size and background color
     width, height = 800, 400
     background_color = "lightblue"
     
@@ -47,7 +37,6 @@ def generate_horoscope_image(text):
     
     # Set font and text color
     try:
-        # Use a default PIL font if no custom font is available
         font = ImageFont.truetype("arial.ttf", size=20)
     except IOError:
         font = ImageFont.load_default()
@@ -64,7 +53,7 @@ def generate_horoscope_image(text):
             line += (words.pop(0) + ' ')
         lines.append(line)
     
-    # Calculate text position for centered alignment
+    # Calculate vertical position for centered text
     total_text_height = sum([draw.textbbox((0, 0), line, font=font)[3] - draw.textbbox((0, 0), line, font=font)[1] for line in lines])
     y_text = (height - total_text_height) // 2
     
@@ -75,7 +64,7 @@ def generate_horoscope_image(text):
         draw.text((x_text, y_text), line, font=font, fill=text_color)
         y_text += draw.textbbox((0, 0), line, font=font)[3] - draw.textbbox((0, 0), line, font=font)[1]
     
-    # Save image to file
+    # Save image to a temporary path
     image_path = "/tmp/horoscope_image.png"
     image.save(image_path)
     return image_path
@@ -104,16 +93,22 @@ with gr.Blocks() as demo:
     # Button to trigger the API call
     btn_get_horoscope = gr.Button("Get Horoscope")
     
-    # Define the click event
+    # Define the button click event
     def on_click(sign, period):
-        horoscope_text, twitter_share_url = get_horoscope(sign, period)
+        horoscope_text = get_horoscope(sign, period)
+        
+        # Prepare Twitter share link
+        share_text = f"Here's my horoscope for {sign.capitalize()} on {period.capitalize()}:\n{horoscope_text}"
+        encoded_text = urllib.parse.quote(share_text)
+        twitter_share_url = f"https://twitter.com/intent/tweet?text={encoded_text}"
         twitter_button_html = f'<a href="{twitter_share_url}" target="_blank" style="color: white; background-color: #1DA1F2; padding: 10px; border-radius: 5px; text-decoration: none;">Share on Twitter</a>'
         
-        # Generate horoscope image and return path for download
+        # Generate horoscope image for download
         image_path = generate_horoscope_image(horoscope_text)
+        
         return horoscope_text, twitter_button_html, image_path
     
-    # Set up click action to update outputs
+    # Set up Gradio click action
     btn_get_horoscope.click(fn=on_click, inputs=[sign_dropdown, period_dropdown], outputs=[horoscope_output, twitter_link, horoscope_image])
 
 if __name__ == "__main__":