Upload app.py with huggingface_hub

app.py

@@ -9,18 +9,25 @@ generator = None
 def initialize_model():
     global generator
     try:
-        # Start with Qwen 2.5-7B-Instruct (more reliable than Omni for Spaces)
-        device = 0 if torch.cuda.is_available() else -1
-        generator = pipeline(
-            "text-generation",
-            model="Qwen/Qwen2.5-7B-Instruct",
-            device=device,
-            torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
-            trust_remote_code=True
-        )
-        return f"Qwen 2.5-7B-Instruct loaded successfully on {'GPU' if device == 0 else 'CPU'}!"
+        # Check available memory and choose model accordingly
+        import psutil
+        available_memory_gb = psutil.virtual_memory().available / (1024**3)
+        
+        if available_memory_gb > 20:  # If we have enough memory, try 7B
+            device = 0 if torch.cuda.is_available() else -1
+            generator = pipeline(
+                "text-generation",
+                model="Qwen/Qwen2.5-7B-Instruct",
+                device=device,
+                torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
+                trust_remote_code=True
+            )
+            return f"Qwen 2.5-7B-Instruct loaded successfully on {'GPU' if device == 0 else 'CPU'}!"
+        else:
+            raise Exception("Insufficient memory for 7B model, falling back to 1.5B")
+            
     except Exception as e:
-        # Fallback to 1.5B model
+        # Fallback to 1.5B model for free tier compatibility
         try:
             generator = pipeline(
                 "text-generation", 
@@ -29,7 +36,7 @@ def initialize_model():
                 torch_dtype=torch.float32,
                 trust_remote_code=True
             )
-            return "Fallback model (Qwen 2.5-1.5B-Instruct) loaded on CPU!"
+            return "Qwen 2.5-1.5B-Instruct loaded successfully on CPU (optimized for free tier)!"
         except Exception as e2:
             # Final fallback to a very reliable model
             try:
@@ -39,7 +46,7 @@ def initialize_model():
                     device=-1,
                     torch_dtype=torch.float32
                 )
-                return "Final fallback model (DialoGPT-large) loaded on CPU!"
+                return "DialoGPT-large loaded as final fallback!"
             except Exception as e3:
                 return f"All models failed: 7B: {str(e)}, 1.5B: {str(e2)}, DialoGPT: {str(e3)}"
 
@@ -51,26 +58,31 @@ def generate_onepager(topic, target_audience, key_points, tone, length):
     length_tokens = {"Short": 200, "Medium": 400, "Long": 600}
     max_tokens = length_tokens.get(length, 400)
     
-    # Create an optimized prompt for Qwen 2.5 instruction format
+    # Create an optimized prompt for actual one-pager format
     prompt = f"""<|im_start|>system
-You are a professional document writer specializing in creating concise, well-structured one-page business documents.
+You are a business consultant who creates compelling one-page documents that drive decisions and action. Your one-pagers are visual, concise, and immediately impactful.
 <|im_end|>
 <|im_start|>user
-Create a professional one-page document about "{topic}" targeted at {target_audience}.
+Create a compelling one-page business document about "{topic}" for {target_audience}.
 
-Requirements:
-- Tone: {tone.lower()}
-- Key points to include: {key_points}
-- Length: {length}
-- Format: Use clear headers and bullet points
-- Structure: Title, Executive Summary, Key Points, Benefits, Recommendations, Conclusion
+Style: {tone.lower()} but action-oriented
+Key points: {key_points}
+Length: {length}
 
-Please write the complete one-page document now.
+Format as a TRUE one-pager with:
+- Bold title and clear audience
+- Visual elements (bullets, emojis, sections)
+- Concise benefits and value proposition  
+- Clear next steps and decision points
+- Urgency and compelling call to action
+
+Make it scannable, impactful, and designed to get approval/buy-in.
 <|im_end|>
 <|im_start|>assistant
-# {topic}
+{topic.upper()}
+{'=' * len(topic)}
 
-## Executive Summary
+🎯 FOR: {target_audience.title()}
 
 """
     
@@ -106,48 +118,39 @@ Please write the complete one-page document now.
 def create_structured_onepager(topic, target_audience, key_points, tone):
     """Create a structured one-pager when AI generation fails"""
     
-    tone_styles = {
-        "Professional": "formal and business-oriented",
-        "Casual": "friendly and approachable",
-        "Academic": "scholarly and research-focused", 
-        "Persuasive": "compelling and action-oriented",
-        "Informative": "clear and educational"
-    }
-    
-    style_desc = tone_styles.get(tone, "professional")
+    key_points_list = [point.strip() for point in key_points.split(',') if point.strip()]
     
-    template = f"""# {topic}
-
-## Executive Summary
-This document provides a comprehensive overview of {topic.lower()} tailored for {target_audience.lower()}. The content is presented in a {style_desc} manner to ensure maximum impact and understanding.
-
-## Key Points
+    # Create a more concise, visual one-pager format
+    template = f"""{topic.upper()}
+{'=' * len(topic)}
 
-{chr(10).join([f"• {point.strip()}" for point in key_points.split(',') if point.strip()])}
+🎯 FOR: {target_audience.title()}
 
-## Background
-{topic} represents an important area that requires careful consideration and strategic thinking. Understanding the core concepts and implications is essential for {target_audience.lower()}.
+💡 WHAT IS IT?
+{topic} is a strategic initiative that delivers measurable value through focused implementation and clear outcomes.
 
-## Main Content
-The fundamental aspects of {topic.lower()} encompass several critical areas that directly impact stakeholders. These elements work together to create a comprehensive framework for understanding and implementation.
+🔑 KEY BENEFITS:
+{chr(10).join([f"   ✓ {point.strip()}" for point in key_points_list[:4]])}
 
-## Benefits & Opportunities
-- Enhanced understanding of core concepts
-- Improved decision-making capabilities  
-- Strategic advantages for implementation
-- Clear actionable insights
+📊 THE OPPORTUNITY:
+• Immediate impact on core business objectives
+• Scalable solution with long-term benefits  
+• Competitive advantage in the market
+• Risk mitigation and operational efficiency
 
-## Recommendations
-1. Begin with thorough analysis of current situation
-2. Develop comprehensive implementation strategy
-3. Monitor progress and adjust approach as needed
-4. Measure results and iterate for continuous improvement
+🚀 NEXT STEPS:
+1. ASSESS current state and requirements
+2. PLAN implementation strategy and timeline  
+3. EXECUTE with clear milestones and metrics
+4. MEASURE results and optimize performance
 
-## Conclusion
-{topic} offers significant opportunities for {target_audience.lower()} when approached strategically. The key points outlined above provide a solid foundation for moving forward with confidence and clarity.
+💰 INVESTMENT: Minimal upfront cost for maximum ROI
+⏱️ TIMELINE: Quick wins in 30-60 days, full benefits within 6 months
+📈 SUCCESS METRICS: Improved efficiency, reduced costs, enhanced outcomes
 
----
-*This one-pager was generated to provide quick, actionable insights on {topic.lower()}.*"""
+────────────────────────────────────────────────────────────────
+DECISION POINT: Move forward with {topic.lower()} to capture these benefits
+Contact: [Your Team] for implementation planning and support"""
     
     return template
 
@@ -155,7 +158,7 @@ The fundamental aspects of {topic.lower()} encompass several critical areas that
 def create_interface():
     with gr.Blocks(title="One-Pager Generator", theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📄 AI One-Pager Generator")
-        gr.Markdown("Generate professional one-page documents on any topic using AI!")
+        gr.Markdown("Generate compelling, decision-focused one-pagers that drive action and get buy-in!")
         
         with gr.Row():
             with gr.Column(scale=1):