app.py

import gradio as gr
import os
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
import torch
import requests
import json
from datetime import datetime

class FinanceAgent:
    def __init__(self):
        """Initialize the Finance Agent with a language model"""
        try:
            # Use a lightweight model that works well for text generation
            model_name = "microsoft/DialoGPT-medium"
            self.tokenizer = AutoTokenizer.from_pretrained(model_name)
            self.model = AutoModelForCausalLM.from_pretrained(model_name)
            
            # Add padding token if it doesn't exist
            if self.tokenizer.pad_token is None:
                self.tokenizer.pad_token = self.tokenizer.eos_token
                
            self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
            self.model.to(self.device)
            
        except Exception as e:
            print(f"Error initializing model: {e}")
            self.model = None
            self.tokenizer = None
    
    def get_finance_knowledge_base(self):
        """Finance knowledge base for common topics"""
        return {
            "stocks": {
                "definition": "Stocks represent ownership shares in a company",
                "key_points": [
                    "Stock prices fluctuate based on supply and demand",
                    "Dividends provide regular income to shareholders",
                    "Growth stocks focus on capital appreciation",
                    "Value stocks are typically undervalued by the market",
                    "Diversification reduces investment risk"
                ]
            },
            "bonds": {
                "definition": "Bonds are debt securities issued by corporations or governments",
                "key_points": [
                    "Bonds pay fixed interest over time",
                    "Government bonds are generally safer than corporate bonds",
                    "Bond prices move inversely to interest rates",
                    "Credit rating affects bond yields",
                    "Bonds provide portfolio stability"
                ]
            },
            "cryptocurrency": {
                "definition": "Digital currencies secured by cryptography",
                "key_points": [
                    "Bitcoin was the first cryptocurrency",
                    "Blockchain technology ensures security",
                    "High volatility characterizes crypto markets",
                    "Regulatory uncertainty affects prices",
                    "Limited supply drives scarcity value"
                ]
            },
            "investing": {
                "definition": "Allocating money to generate returns over time",
                "key_points": [
                    "Start early to benefit from compound interest",
                    "Diversify across asset classes",
                    "Consider risk tolerance and time horizon",
                    "Regular investing through dollar-cost averaging",
                    "Rebalance portfolio periodically"
                ]
            },
            "retirement": {
                "definition": "Planning for financial security after working years",
                "key_points": [
                    "401(k) plans offer tax advantages",
                    "IRA accounts provide additional savings options",
                    "Social Security provides base income",
                    "Healthcare costs increase in retirement",
                    "Estate planning protects beneficiaries"
                ]
            },
            "budgeting": {
                "definition": "Planning and controlling personal expenses",
                "key_points": [
                    "Track income and expenses regularly",
                    "Follow the 50/30/20 rule for allocation",
                    "Build an emergency fund",
                    "Automate savings and bill payments",
                    "Review and adjust budget monthly"
                ]
            }
        }
    
    def generate_finance_response(self, user_input):
        """Generate a comprehensive finance response"""
        user_input_lower = user_input.lower()
        knowledge_base = self.get_finance_knowledge_base()
        
        # Find relevant topics
        relevant_topics = []
        for topic, info in knowledge_base.items():
            if topic in user_input_lower or any(keyword in user_input_lower for keyword in [topic]):
                relevant_topics.append((topic, info))
        
        # Generate response
        if relevant_topics:
            response = f"## Finance Topic: {user_input}\n\n"
            
            for topic, info in relevant_topics:
                response += f"### {topic.title()}\n"
                response += f"**Definition:** {info['definition']}\n\n"
                response += "**Key Points:**\n"
                for point in info['key_points']:
                    response += f"• {point}\n"
                response += "\n"
        else:
            # General finance advice
            response = f"## Finance Topic: {user_input}\n\n"
            response += self.generate_general_advice(user_input)
        
        # Add disclaimer
        response += "\n---\n"
        response += "*Disclaimer: This information is for educational purposes only and should not be considered as financial advice. Please consult with a qualified financial advisor for personalized guidance.*"
        
        return response
    
    def generate_general_advice(self, topic):
        """Generate general financial advice for unknown topics"""
        general_tips = [
            "Consider your financial goals and risk tolerance",
            "Diversification is key to managing investment risk",
            "Start with fundamental analysis before making decisions",
            "Keep informed about market trends and economic indicators",
            "Consider consulting with financial professionals",
            "Maintain an emergency fund for unexpected expenses",
            "Regular review and adjustment of financial strategies is important"
        ]
        
        response = f"Here's some general guidance regarding '{topic}':\n\n"
        response += "**General Financial Principles:**\n"
        for tip in general_tips[:5]:  # Show first 5 tips
            response += f"• {tip}\n"
        
        response += f"\n**Specific to '{topic}':**\n"
        response += f"• Research thoroughly before making any financial decisions related to {topic}\n"
        response += f"• Understand the risks and potential returns associated with {topic}\n"
        response += f"• Consider how {topic} fits into your overall financial portfolio\n"
        
        return response

# Initialize the finance agent
finance_agent = FinanceAgent()

def process_finance_query(user_input, history):
    """Process user input and return finance advice"""
    if not user_input.strip():
        return "Please enter a finance topic or question.", history
    
    try:
        # Generate response
        response = finance_agent.generate_finance_response(user_input)
        
        # Update history
        history.append([user_input, response])
        
        return "", history
        
    except Exception as e:
        error_response = f"I apologize, but I encountered an error processing your request: {str(e)}\n\nPlease try rephrasing your question."
        history.append([user_input, error_response])
        return "", history

def clear_conversation():
    """Clear the conversation history"""
    return [], ""

# Create Gradio interface
with gr.Blocks(
    theme=gr.themes.Soft(),
    title="Finance Agent",
    css="""
    .container {
        max-width: 800px;
        margin: 0 auto;
    }
    .header {
        text-align: center;
        padding: 20px;
        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
        color: white;
        border-radius: 10px;
        margin-bottom: 20px;
    }
    """
) as demo:
    
    gr.HTML("""
    <div class="header">
        <h1>🏦 Finance Agent</h1>
        <p>Your AI-powered financial advisor for quick insights and guidance</p>
    </div>
    """)
    
    with gr.Row():
        with gr.Column(scale=1):
            gr.HTML("""
            <div style="background: #f8f9fa; padding: 15px; border-radius: 10px; margin-bottom: 15px;">
                <h3>💡 How to Use:</h3>
                <ul>
                    <li>Enter any finance topic (stocks, bonds, investing, etc.)</li>
                    <li>Ask questions about budgeting, retirement planning</li>
                    <li>Get quick insights on cryptocurrency, trading</li>
                    <li>Learn about financial planning strategies</li>
                </ul>
            </div>
            """)
            
            gr.HTML("""
            <div style="background: #e8f5e8; padding: 15px; border-radius: 10px;">
                <h3>📈 Popular Topics:</h3>
                <p><strong>Investing:</strong> Stocks, Bonds, ETFs</p>
                <p><strong>Planning:</strong> Retirement, Budgeting</p>
                <p><strong>Trading:</strong> Cryptocurrency, Options</p>
                <p><strong>Banking:</strong> Savings, Loans, Credit</p>
            </div>
            """)
    
    with gr.Column(scale=2):
        chatbot = gr.Chatbot(
            height=500,
            show_label=False,
            container=True,
            bubble_full_width=False
        )
        
        with gr.Row():
            user_input = gr.Textbox(
                placeholder="Enter your finance topic or question (e.g., 'stocks', 'retirement planning', 'cryptocurrency')",
                show_label=False,
                container=False,
                scale=4
            )
            submit_btn = gr.Button("Ask Agent", variant="primary", scale=1)
        
        with gr.Row():
            clear_btn = gr.Button("Clear Chat", variant="secondary")
            gr.HTML("<div style='flex: 1;'></div>")  # Spacer
    
    # Example questions
    gr.Examples(
        examples=[
            ["What are stocks and how do they work?"],
            ["How should I start investing as a beginner?"],
            ["Explain cryptocurrency basics"],
            ["What's the best budgeting strategy?"],
            ["How to plan for retirement?"],
            ["What are the differences between stocks and bonds?"]
        ],
        inputs=user_input,
        label="Example Questions"
    )
    
    # Event handlers
    submit_btn.click(
        process_finance_query,
        inputs=[user_input, chatbot],
        outputs=[user_input, chatbot]
    )
    
    user_input.submit(
        process_finance_query,
        inputs=[user_input, chatbot],
        outputs=[user_input, chatbot]
    )
    
    clear_btn.click(
        clear_conversation,
        outputs=[chatbot, user_input]
    )
    
    gr.HTML("""
    <div style="text-align: center; margin-top: 20px; padding: 15px; background: #f1f3f4; border-radius: 10px;">
        <p><strong>⚠️ Important Disclaimer:</strong> This AI agent provides educational information only. 
        Always consult qualified financial advisors for personalized advice.</p>
        <p>Built with ❤️ using Gradio and Hugging Face</p>
    </div>
    """)

# Launch configuration for Hugging Face Spaces
if __name__ == "__main__":
    demo.launch(
        server_name="0.0.0.0",
        server_port=7860,
        share=False
    )