Update app.py

app.py

@@ -1,14 +1,15 @@
 import gradio as gr
 import requests
+import urllib.parse
+from PIL import Image, ImageDraw, ImageFont
 
 # RapidAPI Credentials and API Endpoint
 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
+# Function to Fetch Horoscope and Create Twitter Share Link
 def get_horoscope(sign, period="today"):
-    # Construct the URL based on the selected sign and period
     url = API_URL_TEMPLATE.format(sign=sign, period=period)
     
     headers = {
@@ -16,7 +17,6 @@ def get_horoscope(sign, period="today"):
         "x-rapidapi-host": API_HOST
     }
     
-    # Send GET request to the API
     try:
         response = requests.get(url, headers=headers)
         response.raise_for_status()
@@ -24,25 +24,97 @@ def get_horoscope(sign, period="today"):
         # Parse JSON response and retrieve the horoscope text
         data = response.json()
         horoscope_text = data.get("general", ["No horoscope available"])[0]
-        return horoscope_text
+        
+        # Prepare the Twitter share link
+        share_text = f"Here's my horoscope by Enemy AI 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}"
+        return f"Error retrieving horoscope: {e}", None
+
+# Function to Generate Image from Horoscope Text
+def generate_horoscope_image(text):
+    # Define image size and background color
+    width, height = 800, 400
+    background_color = "lightblue"
+    
+    # Create a blank image with background color
+    image = Image.new("RGB", (width, height), color=background_color)
+    draw = ImageDraw.Draw(image)
+    
+    # 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()
+    
+    text_color = "black"
+    padding = 20
+    
+    # Wrap text to fit within the image width
+    lines = []
+    words = text.split()
+    while words:
+        line = ''
+        while words and (draw.textsize(line + words[0], font=font)[0] <= width - padding * 2):
+            line += (words.pop(0) + ' ')
+        lines.append(line)
+    
+    # Calculate text position for centered alignment
+    total_text_height = sum([draw.textsize(line, font=font)[1] for line in lines])
+    y_text = (height - total_text_height) // 2
+    
+    # Draw each line of text
+    for line in lines:
+        text_width, text_height = draw.textsize(line, font=font)
+        x_text = (width - text_width) // 2  # Center align
+        draw.text((x_text, y_text), line, font=font, fill=text_color)
+        y_text += text_height
+    
+    # Save image to file
+    image_path = "/tmp/horoscope_image.png"
+    image.save(image_path)
+    return image_path
 
 # Gradio Interface Setup
 with gr.Blocks() as demo:
     gr.Markdown("<center><h1>Daily Horoscope by Enemy AI</h1></center>")
     gr.Markdown("Select your zodiac sign and period to receive your personalized horoscope.")
     
+    # Input dropdowns for sign and period
     sign_dropdown = gr.Dropdown(label="Select Your Zodiac Sign", choices=[
         "aries", "taurus", "gemini", "cancer", "leo", "virgo", 
         "libra", "scorpio", "sagittarius", "capricorn", "aquarius", "pisces"
     ])
     period_dropdown = gr.Dropdown(label="Select Period", choices=["today", "tomorrow", "yesterday"], value="today")
+    
+    # Textbox to display the horoscope
     horoscope_output = gr.Textbox(label="Your Horoscope")
     
+    # Link to share on Twitter
+    twitter_link = gr.HTML()
+    
+    # Image output for download
+    horoscope_image = gr.Image(label="Downloadable Horoscope Image")
+    
     # Button to trigger the API call
     btn_get_horoscope = gr.Button("Get Horoscope")
-    btn_get_horoscope.click(fn=get_horoscope, inputs=[sign_dropdown, period_dropdown], outputs=horoscope_output)
+    
+    # Define the click event
+    def on_click(sign, period):
+        horoscope_text, twitter_share_url = get_horoscope(sign, period)
+        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
+        image_path = generate_horoscope_image(horoscope_text)
+        return horoscope_text, twitter_button_html, image_path
+    
+    # Set up click action to update outputs
+    btn_get_horoscope.click(fn=on_click, inputs=[sign_dropdown, period_dropdown], outputs=[horoscope_output, twitter_link, horoscope_image])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0")